package homework0115;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Scanner;

public class StudentGroupManager {
    public static void main(String[] args) {
        // (1) 创建HashMap对象，泛型为<String, ArrayList<String>>
        HashMap<String, ArrayList<String>> groupMap = new HashMap<>();

        // (2) 存储咱们班每组学员信息，组长姓名为key，组员们的姓名（包括组长自己）为value
        // 这里我们手动添加一些示例数据，你可以根据实际情况从键盘输入
        ArrayList<String> group1 = new ArrayList<>();
        group1.add("组长A");
        group1.add("组员B");
        group1.add("组员C");

        ArrayList<String> group2 = new ArrayList<>();
        group2.add("组长D");
        group2.add("组员E");
        group2.add("组员F");

        groupMap.put("组长A", group1);
        groupMap.put("组长D", group2);

        // 假设我们从键盘输入数据（这里为了简化，直接展示遍历和查找功能）
        // 你可以将下面的数据输入部分替换为实际的键盘输入逻辑

        // (2) 遍历显示每一个小组信息
        System.out.println("每个小组的信息：");
        for (String groupName : groupMap.keySet()) {
            ArrayList<String> members = groupMap.get(groupName);
            System.out.println("组长: " + groupName + ", 组员: " + members);
        }

        // (3) 从键盘输入一个学员姓名，查找这个学员是否咱们班
        Scanner scanner = new Scanner(System.in);
        System.out.println("请输入要查找的学员姓名：");
        String studentName = scanner.nextLine();

        boolean found = false;
        for (ArrayList<String> members : groupMap.values()) {
            if (members.contains(studentName)) {
                found = true;
                System.out.println("学员 " + studentName + " 在咱们班，属于组长 " + groupMap.get(members.get(0)).get(0) + " 的小组。");
                break; // 如果只需要找到第一个匹配项，就跳出循环
            }
        }

        if (!found) {
            System.out.println("学员 " + studentName + " 不在咱们班。");
        }

        // 关闭Scanner对象
        scanner.close();
    }
}
